/*
 *  Copyright 2009 Advanced Industrial Science and Technology (AIST).
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

/*! \file
 *  \brief
 */
#pragma once

#include <thrust/graph/detail/out_edge_iterator.hpp>
#include <thrust/iterator/iterator_adaptor.h>

namespace thrust {

// forward declaration on out_edge_iterator
template <typename OutEdgeIter
          , typename VertexDescriptor
          , typename EdgeDescriptor
          , typename CalledSpace>
struct out_edge_iter;


// forward declaration on in_edge_iterator
template <typename InEdgeIter
          , typename VertexDescriptor
          , typename EdgeDescriptor
          , typename CalledSpace>
struct in_edge_iter;

namespace detail {

template <typename OutEdgeIter // OutEdgeIter is device_ptr<StoredEdge>
          , typename VertexDescriptor
          , typename EdgeDescriptor
          , typename CalledSpace>
struct out_edge_iter_base
{
  typedef OutEdgeIter    base_iterator;
  typedef EdgeDescriptor value_type;
  typedef value_type     reference;
  typedef value_type     pointer;
  typedef typename iterator_space<base_iterator>::type space;
  typedef typename iterator_traversal<base_iterator>::type traversal;

public:
  typedef typename thrust::experimental::iterator_adaptor<
  out_edge_iter<OutEdgeIter, VertexDescriptor, EdgeDescriptor, CalledSpace> //self
  , base_iterator  // Base
  , pointer        // Pointer
  , value_type     // Value
  , space          // Space
  , traversal      // Traversal
  , reference      // Reference
  //, use_default    // Difference
  > type;
};  

template <typename InEdgeIter // InEdgeIter is device_ptr<StoredEdge>
          , typename VertexDescriptor
          , typename EdgeDescriptor
          , typename CalledSpace>
struct in_edge_iter_base
{
  typedef InEdgeIter     base_iterator;
  typedef EdgeDescriptor value_type;
  typedef value_type     reference;
  typedef value_type     pointer;
  typedef typename iterator_space<base_iterator>::type space;
  typedef typename iterator_traversal<base_iterator>::type traversal;

public:
  typedef typename thrust::experimental::iterator_adaptor<
  in_edge_iter<InEdgeIter, VertexDescriptor, EdgeDescriptor, CalledSpace> //self
  , base_iterator  // Base
  , pointer        // Pointer
  , value_type     // Value
  , space          // Space
  , traversal      // Traversal
  , reference      // Reference
  //, use_default    // Difference
  > type;
};

} // end namespace detail

} // end namespace thrust
